home *** CD-ROM | disk | FTP | other *** search
/ Personal Computer World 2009 February / PCWFEB09.iso / Software / Linux / Kubuntu 8.10 / kubuntu-8.10-desktop-i386.iso / casper / filesystem.squashfs / usr / share / hplip / systray.py < prev    next >
Text File  |  2008-10-13  |  5KB  |  177 lines
  1. #!/usr/bin/env python
  2. # -*- coding: utf-8 -*-
  3. #
  4. # (c) Copyright 2003-2007 Hewlett-Packard Development Company, L.P.
  5. #
  6. # This program is free software; you can redistribute it and/or modify
  7. # it under the terms of the GNU General Public License as published by
  8. # the Free Software Foundation; either version 2 of the License, or
  9. # (at your option) any later version.
  10. #
  11. # This program is distributed in the hope that it will be useful,
  12. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14. # GNU General Public License for more details.
  15. #
  16. # You should have received a copy of the GNU General Public License
  17. # along with this program; if not, write to the Free Software
  18. # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  19. #
  20. # Author: Don Welch
  21.  
  22. __version__ = '0.1'
  23. __title__ = 'System Tray Status Service'
  24. __doc__ = ""
  25.  
  26. # StdLib
  27. import sys
  28. import os
  29. import getopt
  30. import signal
  31.  
  32. # Local
  33. from base.g import *
  34. from base import utils
  35. from prnt import cups
  36.  
  37.  
  38. USAGE = [(__doc__, "", "name", True),
  39.          ("Usage: hp-systray [OPTIONS]", "", "summary", True),
  40.          utils.USAGE_OPTIONS,
  41.          ("Force Qt3:", "--qt3 (default)", "option", False),
  42.          ("Force Qt4:", "--qt4", "option", False),
  43.          ("Startup even if no hplip CUPS queues are present:", "-x or --force-startup", "option", False),
  44.          utils.USAGE_LOGGING1, utils.USAGE_LOGGING2, utils.USAGE_LOGGING3,
  45.          utils.USAGE_HELP,
  46.         ]
  47.  
  48.  
  49. def usage(typ='text'):
  50.     if typ == 'text':
  51.         utils.log_title(__title__, __version__)
  52.  
  53.     utils.format_text(USAGE, typ, __title__, 'hpssd.py', __version__)
  54.     sys.exit(0)
  55.  
  56.  
  57.  
  58. if __name__ == '__main__':
  59.     log.set_module('hp-systray(init)')
  60.  
  61.     prop.prog = sys.argv[0]
  62.     force_qt3 = False
  63.     force_qt4 = False
  64.     force_startup = False
  65.  
  66.     try:
  67.         opts, args = getopt.getopt(sys.argv[1:], 'l:hgx', 
  68.             ['level=', 'help', 'help-man', 'help-rest', 'help-desc',
  69.             'qt3', 'qt4', 'force-startup'])
  70.  
  71.     except getopt.GetoptError, e:
  72.         log.error(e.msg)
  73.         usage()
  74.  
  75.     if os.getenv("HPLIP_DEBUG"):
  76.         log.set_level('debug')
  77.  
  78.     for o, a in opts:
  79.         if o in ('-l', '--logging'):
  80.             log_level = a.lower().strip()
  81.             if not log.set_level(log_level):
  82.                 usage()
  83.  
  84.         elif o == '-g':
  85.             log.set_level('debug')
  86.  
  87.         elif o in ('-h', '--help'):
  88.             usage()
  89.  
  90.         elif o == '--help-rest':
  91.             usage('rest')
  92.  
  93.         elif o == '--help-man':
  94.             usage('man')
  95.  
  96.         elif o == '--help-desc':
  97.             print __doc__,
  98.             sys.exit(0)
  99.  
  100.         elif o == '--qt3':
  101.             force_qt3 = True
  102.             force_qt4 = False
  103.  
  104.         elif o == '--qt4':
  105.             force_qt4 = True
  106.             force_qt3 = False
  107.             
  108.         elif o in ('-x', '--force-startup'):
  109.             force_startup = True
  110.  
  111.  
  112.     utils.log_title(__title__, __version__) 
  113.     
  114.     if os.getuid() == 0:
  115.         log.error("hp-systray cannot be run as root. Exiting.")
  116.         sys.exit(1)
  117.  
  118.     if not force_startup:
  119.         # Check for any hp: or hpfax: queues. If none, exit
  120.         if not utils.any([p.device_uri for p in cups.getPrinters()], lambda x : x.startswith('hp')):
  121.             log.warn("No hp: or hpfax: devices found in any installed CUPS queue. Exiting.")
  122.             sys.exit(1)
  123.     
  124.     ok, lock_file = utils.lock_app('hp-systray')
  125.     if not ok:
  126.         sys.exit(1)
  127.     
  128.     r, w = os.pipe()
  129.     parent_pid = os.getpid()
  130.     log.debug("Parent PID=%d" % parent_pid)
  131.     child_pid = os.fork()
  132.  
  133.     if child_pid:
  134.         # parent (UI)
  135.         os.close(w)
  136.  
  137.         try:
  138.             if force_qt3 or (not force_qt3 and not force_qt4):
  139.                 from ui import systemtray_qt3
  140.                 systemtray_qt3.run(r, child_pid)
  141.     
  142.             elif force_qt4:
  143.                 from ui import systemtray_qt4
  144.                 systemtray_qt4.run(r, child_pid)
  145.  
  146.         except: pass
  147.         
  148.         finally:
  149.             if child_pid:
  150.                 log.debug("Killing child systray process (pid=%d)..." % child_pid)
  151.                 try:
  152.                     os.kill(child_pid, signal.SIGKILL)
  153.                 except OSError, e:
  154.                     log.debug("Failed: %s" % e.message)
  155.                 
  156.             utils.unlock(lock_file)
  157.  
  158.                 
  159.     else:
  160.         # child (dbus)
  161.         os.close(r)
  162.  
  163.         try:
  164.             import hpssd
  165.             hpssd.run(w, parent_pid)
  166.         finally:
  167.             if parent_pid:
  168.                 log.debug("Killing parent systray process (pid=%d)..." % parent_pid)
  169.                 try:
  170.                     os.kill(parent_pid, signal.SIGKILL)
  171.                 except OSError, e:
  172.                     log.debug("Failed: %s" % e.message)
  173.  
  174.             utils.unlock(lock_file)
  175.     
  176.  
  177.